import DatabaseSelfCheck as DSC
import GetData as GD
import time
import random
import configparser
import Display
#配置文件地址
config_path = './config/config.ini'
config = configparser.ConfigParser()
config.read(config_path)
sleep_or_not = config.get('NETWORK','sleep')
if sleep_or_not == 'True':
    sleep_or_not = 1
else:
    sleep_or_not = 0
sleep_or_not = bool(sleep_or_not)

mode = config.get('TEXTMODE','MODE')
if mode == 'True':
    mode = 1
else:
    mode = 0
mode = bool(mode)

def __SLEEP__():
    if sleep_or_not:
        time.sleep(random.uniform(3,6))
    else:
        pass

def start():
    Display.__display__()
    DSC.start_check()
    if not GD.get_data_cpu_url():
        print('无法获取CPU数据')
        print('无法正常获取，请更换IP地址或者增加请求头信息')
    else:
        print('爬取完成：CPU，耐心等待..')

    if not mode:
        __SLEEP__()
        if not GD.get_data_mainboard_url():
            print('无法获取主板数据')
            print('无法正常获取，请更换IP地址或者增加请求头信息')
        else:
            print('爬取完成：主板，耐心等待..')
        __SLEEP__()

        if not GD.get_data_GraphicsCard_url():
            print('无法获取显卡数据')
            print('无法正常获取，请更换IP地址或者增加请求头信息')
        else:
            print('爬取完成：显卡，耐心等待..')
        __SLEEP__()

        if not GD.get_data_Memory_url():
            print('无法获取内存数据')
            print('多次尝试可能导致IP被封锁！！！')
        else:
            print('爬取完成：内存，耐心等待..')
        __SLEEP__()

        if not GD.get_data_HardDisk_url():
            print('无法获取硬盘数据')
            print('多次尝试可能导致IP被封锁！！！')
        else:
            print('爬取完成：硬盘，耐心等待..')

    x = input('已结束爬取，回车结束并释放进程。')